Skip to content

feat(appkit): stream agent model calls through the modular request seam - #641

Draft
MarioCadenas wants to merge 1 commit into
feat/modular-aisearchfrom
feat/modular-agents
Draft

MarioCadenas wants to merge 1 commit into
feat/modular-aisearchfrom
feat/modular-agents

Conversation

@MarioCadenas

Copy link
Copy Markdown
Collaborator

Part of the migration off @databricks/sdk-experimental onto the modular @databricks/sdk-* SDK (see #562). Base: feat/modular-sdk-integration.

Changes

feat(appkit): stream agent model calls through the modular request seam

Serving, AI Gateway and Supervisor API streaming now go through the
workspace client's request() (modular transport, AppKit User-Agent, same
auth resolution incl. OBO token -> PAT) instead of the legacy
apiClient.request({ raw: true }). fromSupervisorApi resolves the host via
getHost() in place of config.ensureResolved().

The public supervisor WorkspaceClientLike stays exported. A legacy SDK
client with only apiClient.request (+ config.ensureResolved) is still
accepted through a deprecated fallback branch, so callers keep working.

Draft: implemented by an agent; needs review and a deployed SP + OBO check before it's marked ready.

This pull request and its description were written by Isaac.

@MarioCadenas
MarioCadenas changed the base branch from feat/modular-sdk-integration to feat/modular-aisearch October 8, 2026 14:38
@MarioCadenas
MarioCadenas added this pull request to stack #636 October 8, 2026 14:38
@github-actions

github-actions Bot commented Oct 8, 2026 •

Copy link
Copy Markdown
Contributor

📦 Bundle size report

Compared against bundle-size-baseline.json (main).

@databricks/appkit

npm tarball (packed): 1.2 MB (+23 KB) — gzipped download (dist + bin; excludes release-only docs/NOTICE).

dist raw gzip
JS (runtime) 1.3 MB (+22 KB) 456 KB (+8.5 KB)
Type declarations 461 KB (+5.8 KB) 167 KB (+2.1 KB)
Source maps 2.5 MB (+43 KB) 852 KB (+14 KB)
Other 11 KB 3.7 KB
Total 4.2 MB (+70 KB) 1.4 MB (+25 KB)
Per-entry composition (own code — deps external (as shipped))
Entry Initial (gz) Lazy (gz) Total (gz) node_modules (min) Own code (min)
. 104 KB (+2.4 KB) 2.5 KB (+1 B) 107 KB (+2.4 KB) external 339 KB (+7.1 KB)
./beta 100 KB (+1.8 KB) 485 B (+1 B) 101 KB (+1.8 KB) external 302 KB (+4.6 KB)
./testing 43 KB (+1.5 KB) 32 KB (+35 B) 76 KB (+1.5 KB) external 219 KB (+4.3 KB)
./tsdown 520 B 0 B 520 B external 813 B
./type-generator 25 KB (+1.4 KB) 0 B 25 KB (+1.4 KB) external 70 KB (+3.9 KB)

Chunks:

Entry Chunk Load Size (gz)
. index.js initial 100 KB
. utils.js initial 4.6 KB
. remote-tunnel-manager.js lazy 2.5 KB
./beta beta.js initial 82 KB
./beta stream-manager.js initial 5.9 KB
./beta modular.js initial 4.6 KB
./beta service-context.js initial 4.2 KB
./beta databricks.js initial 3.3 KB
./beta client.js initial 657 B
./beta index.js initial 20 B
./beta supervisor-api.js lazy 192 B
./beta databricks.js lazy 178 B
./beta index.js lazy 115 B
./testing manifest.js initial 30 KB
./testing index.js initial 10 KB
./testing wide-event-emitter.js initial 2.9 KB
./testing index.js lazy 28 KB
./testing remote-tunnel-manager.js lazy 2.5 KB
./testing utils.js lazy 1.8 KB
./tsdown index.js initial 520 B
./type-generator index.js initial 25 KB

@databricks/appkit-ui

npm tarball (packed): 397 KB (+572 B) — gzipped download (dist + bin; excludes release-only docs/NOTICE).

dist raw gzip
JS (runtime) 429 KB (+235 B) 147 KB (+108 B)
Type declarations 267 KB (+809 B) 98 KB (+362 B)
Source maps 853 KB (+314 B) 285 KB (+143 B)
CSS 16 KB 3.2 KB
Total 1.5 MB (+1.3 KB) 534 KB (+613 B)
Per-entry composition (consumer bundle — deps bundled, peerDeps external)
Entry Initial (gz) Lazy (gz) Total (gz) node_modules (min) Own code (min)
./js 5.3 KB 49 KB 55 KB 208 KB 14 KB
./js/beta 2.1 KB 0 B 2.1 KB 0 B 5.0 KB
./react 433 KB (+11 B) 49 KB 481 KB (+11 B) 1.3 MB 177 KB (+41 B)
./react/beta 4.9 KB 0 B 4.9 KB 0 B 12 KB

Chunks:

Entry Chunk Load Size (gz)
./js index.js initial 5.2 KB
./js chunk initial 120 B
./js apache-arrow lazy 49 KB
./js/beta beta.js initial 2.1 KB
./react index.js initial 431 KB
./react tslib initial 2.1 KB
./react apache-arrow lazy 49 KB
./react/beta beta.js initial 4.9 KB

@github-actions

github-actions Bot commented Oct 9, 2026 •

Copy link
Copy Markdown
Contributor

🤖 AppKit PR bot

🔬 Run evals

Start an eval for this PR from the evals-monitor app: Go to Evals Monitor →

📦 Try this PR's app template

Scaffolds a new app from this PR's SDK build. Run it in any folder (requires the GitHub CLI — gh auth login — and the Databricks CLI):

gh run download 37949340180 -R databricks/appkit -n appkit-template-0.87.0-pr.3e49268-feat-modular-agents-641 -D appkit-pr-641 \
  && unzip -o "appkit-pr-641/appkit-template-0.87.0-pr.3e49268-feat-modular-agents-641.zip" -d "appkit-pr-641" \
  && databricks apps init --template "appkit-pr-641"

The template pins @databricks/appkit and @databricks/appkit-ui to tarballs built from this branch, so the scaffolded app runs against this PR's code.

Serving, AI Gateway and Supervisor API streaming now go through the
workspace client's request() (modular transport, AppKit User-Agent, same
auth resolution incl. OBO token -> PAT) instead of the legacy
apiClient.request({ raw: true }). fromSupervisorApi resolves the host via
getHost() in place of config.ensureResolved().

The public supervisor WorkspaceClientLike stays exported. A legacy SDK
client with only apiClient.request (+ config.ensureResolved) is still
accepted through a deprecated fallback branch, so callers keep working.

Co-authored-by: Isaac <no-reply@databricks.com>
Signed-off-by: MarioCadenas <MarioCadenas@users.noreply.github.com>

# Conflicts:
#	packages/appkit/src/connectors/serving/client.ts
#	packages/appkit/src/connectors/serving/tests/client.test.ts

This branch has not been deployed

No deployments
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant